Options
All
  • Public
  • Public/Protected
  • All
Menu

Interface TrainingOptions

训练配置

Hierarchy

  • TrainingOptions

Index

Properties

async

async: boolean

是否异步训练

batchSize

batchSize: number

epochs

epochs: number

Optional onBatch

onBatch: (batch: number, size: number, loss: number) => void

Type declaration

    • (batch: number, size: number, loss: number): void
    • Parameters

      • batch: number
      • size: number
      • loss: number

      Returns void

Optional onEpoch

onEpoch: (epoch: number, loss: number) => void

Type declaration

    • (epoch: number, loss: number): void
    • Parameters

      • epoch: number
      • loss: number

      Returns void

Optional onTrainEnd

onTrainEnd: (loss: number) => void

Type declaration

    • (loss: number): void
    • Parameters

      • loss: number

      Returns void

Generated using TypeDoc